# 导入正则模块
import re,random
# 导入常量文件
from info import constants
from flask import request,jsonify ,current_app,make_response,session
# 导入蓝图对象
from . import passport_blue
# 导入captcha 工具里面的对象
from info.utils.captcha.captcha import captcha
# 导入自定义状态码那个类RET
from info.utils.response_code import RET
# 导入redis实例
from info import redis_store,db
# 导入云通讯
from info.libs.yuntongxun.sms import CCP
# 导入模型类
from info.models import User
# 导入时间模块
from datetime import datetime
# 使用蓝图对象


@passport_blue.route('/image_code')
def generate_image_code():
    # 传入参数
    # 生成图片验证码

    # 接口步骤
    # 接受参数----检查参数----业务处理----返回参数
    # 01  获取前段传来的参数 uuid 使用查询字符串来获取  需要使用request对象 获取args
    # 02 检查获取道到的参数

    # 03 如果没有就返回信息
    # 04 如果有参数 就使用captcha 工具生成验证码图片
    # 因为在captcha中返回的是name text  image
    # 我们需要的是图片验证码中的内容text  然后把text根据uuid存在redis中
    # 05 返回参数


    # 获取参数
    image_code_id = request.args.get('image_code_id')
    print(image_code_id)
    # 判断参数是否有值
    if not image_code_id:
        # 返回数据需要使用json
        return jsonify(errno = RET.PARAMERR,errmsg = '参数错误')
    # 使用对象captcha 调用captcha 工具中的函数创建图片验证码
    name,text,image = captcha.generate_captcha()

    # 然后把text根据每次创建的uuid存入到redis中
    #  在info/init文件中 实例化一个redis对象
    # 通过setex方法 键 时间 值
    # 变量名加上uuid是在redis中存储的键  然后设置时间 值为生成的text内容  时间使用常量文件中的常量
    # 操作数据库需要使用try 语句
    try:
        redis_store.setex('ImageCode'+image_code_id,constants.IMAGE_CODE_REDIS_EXPIRES,text)
    except Exception as e:
        # 使用全局对象logger对象
        current_app.logger.error(e)
        # 如果无法存储到数据库中 那么直接终止程序运行
        return jsonify(errno = RET.DATAERR,errmsg = '数据错误')
    else:
        # 如果成功直接返回数据 使用返回对象返回图片
        response = make_response(image)
        return response


@passport_blue.route('/sms_code',methods = ['POST'])
def send_sms_code():


    # 获取参数---检查参数---业务处理---返回参数
    # 01 发送短信需要知道的有 手机号 验证码 也需要正确 验证码 对比需要前段再次把uuid发送过来 然后取redis里面查找
    # 02 检查参数必须都得存在
    # 03 检查手机好的格式 正则表达式
    # 04 检查图片验证码是否正确
    # 05 从redis中获取真是的图片验证吗  不一定能获取到
    # 06 所以需要判断图片获取结果是否存在
    # 07 如果存在 先删除redis中的图片验证码 因为图片验证码不可以比较多次  本质是只能读取redis一次
    # 08 比较图片验证码是否正确
    # 09 生成短信验证码  六位数数值
    # 10 需要存储redis中
    # 11 调用云通讯发送短信
    # 12  保存发送结果
    # 13 返回结果

    # 获取post请求的三个参数
    # 获取电话号码
    mobile = request.json.get('mobile')
    # 获取输入的内容
    image_code = request.json.get('image_code')
    # 获取uuid
    image_code_id = request.json.get('image_code_id')
    # 然后判断三个参数是否为空  一起判断
    if not all([mobile,image_code,image_code_id]):
        return jsonify(errno = RET.DATAERR,errmsg = '数据错误')
    # 检查手机格式 使用正则表达式
    if not re.match(r'1[3456789]\d{9}$',mobile):
        return jsonify(errno = RET.DATAERR,errmsg = '手机号不存在')

    # 如果手机号正确则根据创过来的uuid 从数据库中查找图片内容
    try:
        real_image_code = redis_store.get('ImageCode'+image_code_id)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno = RET.DBERR,errmsg = '查询数据失败')

    # 判断获取到的值  数据库里设置的存储内容有可能过期
    if not real_image_code:
        return jsonify(errno = RET.NODATA,errmsg = '验证码已过期')
    # 如果获取到了图片验证码的内容 需要把redis中的内容删除 因为只能get一次
    try:  # 根据这个键删除这个内容
        redis_store.delete('ImageCode_' + image_code_id)
    except Exception as e:
        current_app.logger.error(e)

    # 比较获取到 的内容 记得忽略大小写
    if real_image_code.lower() != image_code.lower():
        return jsonify(errno = RET.DATAERR,errmsg = '验证码错误')


    # 检查用户是否已经注册 使用模型类来查询
    try:
        # 保存这个查询结果
        user = User.query.filter(User.mobile == mobile).first()
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno = RET.DATAERR,errmsg = '查询用户数据失败')
    else:
        # 如果用户数据不是空
        if user is not None:
            return jsonify(errno = RET.DATAEXIST,errmsg = '用户以注册')




    # 业务处理
    # 生成随机验证码  存储到redis中
    sms_code = '%06d' % random.randint(0,999999)
    try:
        # 将生成到的随机码 存储到 redis中 根据手机号存储僵尸唯一的
        redis_store.setex('SMS'+mobile,constants.SMS_CODE_REDIS_EXPIRES,sms_code)
        print(mobile)
        print(sms_code)
        print(image_code)
    except Exception as e:
        current_app.logger(e)
        return jsonify(errno=RET.DBERR,errmsg='保存数据失败')


    # 通过云通讯 发送随机码
    try:
        # 实例化云通讯对象
        ccp = CCP()
        # 发送 手机号  参数  模板
        result = ccp.send_template_sms(mobile,[sms_code,constants.SMS_CODE_REDIS_EXPIRES/60],1)
    except Exception as e:
        current_app.logger(e)
        return jsonify(errno=RET.THIRDERR, errmsg='发送短信异常')
    # 判断发送结果 因为在云通讯里调用这个函数会有一个返回值
    print(result)
    if result == 0:
        return jsonify(errno=RET.OK, errmsg='发送成功')
    else:
        return jsonify(errno=RET.THIRDERR, errmsg='发送失败')


# 用户注册
@passport_blue.route('/register',methods = ['POST'])
def register():
    # 用户注册流程：
    # 01 接受参数   检查参数  业务处理  返回参数
    # 02 接受的参数有 mobile 手机号  sms——code 短信验证码  password 密码
    # 03 检查参数手机号的格式
    # 04 检查用户输入短信验证码是是否正确
    # 05 尝试从redis中获取真是的短信验证码 根据mobile 获取
    # 06 判断获取结果 如果不存在 表示短信验证码已经过期
    # 07 比较短信验证码是否正确
    # 08 删除已经比较过的redis中的短信验证码 短信验证码是先比较在删除
    # 因为短信验证码可以比较多次
    # 09 需要使用模型类对象 对密码进行加密存储(也可以在用户刚刚输入完电话的时候提示 但是不安全）
    # 10 确认用户未注册根据手机号查询用户
    # 11 保存用户信息 提交数据 如果发生异常 进行回滚
    # 12 缓存用户信息 用户id
    # 13 返回结果


    # 获取参数
    mobile = request.json.get('mobile')
    # 获取手机验证码
    sms_code = request.json.get('sms_code')
    password = request.json.get('password')
    # 检查参数的完整性
    print(mobile)
    print(sms_code)
    print(password)
    if not all([mobile,sms_code,password]):
        return jsonify(errno = RET.PARAMERR,errmsg = '数据不存在')

    # 检查手机号格式 使用正则表达式
    if not re.match(r'1[3456789]\d{9}',mobile):
        return jsonify(errno = RET.PARAMERR,errmsg = '手机号格式不对')

    # 检查短信验证码  是根据上面存导redis中的进行比较
    try:
     real_sms_code = redis_store.get('SMS'+mobile)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno = RET.DATAERR,errmsg = 'chaxun shuju shipai ')
    if not real_sms_code:
        return jsonify(errno = RET.PARAMERR,errmsg = '短信验证码已过期')
    if real_sms_code != str(sms_code):  #有可能是数字 所以转成字符串
        return jsonify(errno = RET.DATAEXIST,errmsg = '短信验证码错误')


    # 如果短信验证码是正确的那马对比过后删除 redis中的短信验证码
    try:
        redis_store.delete('SMS'+mobile)
    except Exception as e:
        current_app.logger.error(e)

    # 根据手机号查询是否注册
        try:
            user = User.query.filter(User.mobile == mobile).first()
        except Exception as e:
            current_app.logger.error(e)
            return jsonify(errno = RET.DBERR,errmsg = '查询数据失败')
        else:
            if user:
                return jsonify(errno = RET.PARAMERR,errmsg = '此手机号已注册')


    # 检查密码  构造模型类对象 存储用户信息 实现密码的加密存储
    # 定义对象 然后存储
    user = User()
    user.mobile = mobile
    # 昵称
    user.nick_name = mobile
    # 密码存储调用了modles中的 def password(self, value):  方法 给password加密
    user.password = password

    # 把用户信息存储到 mysql数据库中
    try:

        db.session.add(user)
        db.session.commit()
    except Exception as e:
        current_app.logger.error(e)
        # 需要回滚
        db.session.rollback()
        return jsonify(errno = RET.DBERR,errmsg = '保存数据失败')


    # 缓存用户信息 实现状态保持 是因为当我们注册成功的时候就应该自动登陆
    # 实现状态保持需要用到 session
    # request.query.
    session['user_id'] = user.id
    session['mobile'] = mobile
    # 昵称
    session['nick_name'] = mobile
    # 返回结果
    print("session存储成功")
    return jsonify(errno = RET.OK,errmsg = 'ok')


# 登陆系统

@passport_blue.route('/login',methods = ['POST'])
def login():

    # 用户登陆分析：
    # 获取参数  检查参数 业务处理  返回参数
    # 01 获取参数  mobile password
    # 02 检查参数 完整性
    # 03 检查手机号的格式
    # 04 根据手机号查询用户信息 查看用户是否注册
    # 05确认查询结果
    # 06 确认密码正确
    # 07 保存用户登陆的时间
    # 08 提交数据到mysql数据库
    # 09 缓存用户信息
    # 因为用户可以登陆多次 有可能修改用户昵称 如果修改昵称 是修改后的结果 如果为修改 默认为手机号


    # 获取参数
    # print(request.json)
    mobile = request.json.get('mobile')
    print(mobile)
    password = request.json.get('password')
    if not all([mobile,password]):
        return jsonify(errno = RET.NODATA,errmsg = '没有数据')
    if not re.match(r'1[3456789]\d{9}$',mobile):
        return jsonify(errno = RET.DATAERR,errmsg = '手机号格式不正确')
    try:
        # 检查手机号是否被注册过 区数据库里面查询
        user = User.query.filter(User.mobile == mobile).first()
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno = RET.DATAERR,errmsg = '数据库错误')
    # if  user is None:
    #     return jsonify(errno = RET.DATAERR,errmsg = '用户尚未注册')

    # 用户名一注册和密码错误 可以放在一起 判断 这样就不可以反推出来 用户名已注册
    # 这个查找导的user对象调用这个函数 把新密码传入进去 然后加密 然后与自己进行对比

    if user is None and not user.check_password(password):
        return jsonify(erron = RET.DATAERR,errmsg = '用户名或密码错误')

    # 保存用户登陆时间
    user.last_login = datetime.now()
    try:
        db.session.add(user)
        db.session.commit()
    except Exception as e:
        current_app.logger(e)
        # 如果提交失败 则需要回滚
        db.session.rollback()
        return jsonify(errno = RET.DBERR,errmsg = '数据保存失败')

    # 缓存用户信息

    session['user_id'] = user.id
    session['mobile'] = mobile
    # 这种方法就是从数据库里面那新的数据给他
    session['nick_name'] = user.nick_name
    return jsonify(errno = RET.OK,errmsg = 'ok')


@passport_blue.route('/logout')
def loggout():
    # 用户退出

    # 用户退出本质就是
    # 删除redis中的缓存数据

    session.pop('user_id',None)
    session.pop('mobile',None)
    session.pop('nick_name',None)
    return jsonify(errno = RET.OK,errmsg = 'ok')
    pass



